YAML processing in python